www.gusucode.com > VC++ 同学录信息管理系统源代码-源码程序 > VC++ 同学录信息管理系统源代码-源码程序/code/DataDlg.cpp

    // DataDlg.cpp : implementation file
// Download by http://www.NewXing.com

#include "stdafx.h"
#include "Mates.h"
#include "DataDlg.h"
#include "AboutDialog.h"
#ifdef _DEBUG
#define new DEBUG_NEW
#undef THIS_FILE
static char THIS_FILE[] = __FILE__;
#endif

/////////////////////////////////////////////////////////////////////////////
// CDataDlg dialog
static UINT indicators[]=
	{
			IDS_STATUS_DATA,
			ID_INDICATOR_CAPS,
			ID_INDICATOR_CLOCK,
			ID_INDICATOR_NUM,		
			ID_INDICATOR_SCRL,
			IDS_STATUS_SPACE,
	};

CDataDlg::CDataDlg(CWnd* pParent /*=NULL*/)
	: CDialog(CDataDlg::IDD, pParent)
{
	//{{AFX_DATA_INIT(CDataDlg)
	m_strMajor = _T("");
	m_strSex = _T("");
	m_strName = _T("");
	m_strNativePlace = _T("");
	m_strUniver = _T("");
	m_timeBirthday = 0;
	m_strTel = _T("");
	//}}AFX_DATA_INIT
	m_TreeBootImage.Create (IDB_BITMAP_TREE,20,1,ILC_COLOR32);  //以20象素为宽截取图标
}


void CDataDlg::DoDataExchange(CDataExchange* pDX)
{
	CDialog::DoDataExchange(pDX);
	//{{AFX_DATA_MAP(CDataDlg)
	DDX_Control(pDX, IDC_EDIT_NAME, m_ctrlName);
	DDX_Control(pDX, IDC_COMBO_MAJOR, m_ctrlMajor);
	DDX_Control(pDX, IDC_TREE, m_ctrlTree);
	DDX_Control(pDX, IDC_LIST, m_ctrlList);
	DDX_CBString(pDX, IDC_COMBO_MAJOR, m_strMajor);
	DDX_CBString(pDX, IDC_COMBO_SEX, m_strSex);
	DDX_Text(pDX, IDC_EDIT_NAME, m_strName);
	DDX_Text(pDX, IDC_EDIT_NATIVEPLACE, m_strNativePlace);
	DDX_Text(pDX, IDC_EDIT_UNIVER, m_strUniver);
	DDX_DateTimeCtrl(pDX, IDC_DATE_BIRTHDAY, m_timeBirthday);
	DDX_Text(pDX, IDC_EDIT_TEL, m_strTel);
	//}}AFX_DATA_MAP
}


BEGIN_MESSAGE_MAP(CDataDlg, CDialog)
	//{{AFX_MSG_MAP(CDataDlg)
	ON_COMMAND(ID_ITEM_REFRESH, OnItemRefresh)
	ON_NOTIFY(TVN_SELCHANGED, IDC_TREE, OnSelchangedTree)
	ON_NOTIFY(NM_CLICK, IDC_LIST, OnClickList)
	ON_NOTIFY(LVN_COLUMNCLICK, IDC_LIST, OnColumnclickList)
	ON_BN_CLICKED(ID_ITEM_SEARCH, OnItemSearch)
	ON_COMMAND(ID_ITEM_CLEAR, OnItemClear)
	ON_BN_CLICKED(ID_ITEM_DELETE, OnItemDelete)
	ON_BN_CLICKED(ID_ITEM_EDIT, OnItemEdit)
	ON_COMMAND(ID_ITEM_HELP, OnItemHelp)
	ON_COMMAND(ID_ITEM_QUIT, OnItemQuit)
	ON_WM_TIMER()
	ON_COMMAND(ID_ITEM_VERSION, OnItemVersion)
	ON_BN_CLICKED(ID_ITEM_ADD, OnItemAdd)
	ON_WM_CONTEXTMENU()
	ON_BN_CLICKED(IDC_BUTTON_BIRTHDAY, OnButtonBirthday)
	//}}AFX_MSG_MAP
END_MESSAGE_MAP()

/////////////////////////////////////////////////////////////////////////////
// CDataDlg message handlers
void CDataDlg::AddTreeNodes(){
	m_ctrlTree.DeleteAllItems();	       
	hRootItem=m_ctrlTree.InsertItem("所在地",0,2,TVI_ROOT,TVI_LAST);
	m_pRS.MoveFirst();		      
	while(!m_pRS.ADOEOF())
	{
		
		HTREEITEM hNative=AddDistinctNode(hRootItem,m_pRS.GetFieldString("StuNativePlace"));
		
		HTREEITEM hUniver=AddDistinctNode(hNative,m_pRS.GetFieldString("StuUniver"));		
		m_pRS.MoveNext();
	}
}


HTREEITEM	CDataDlg::AddDistinctNode(HTREEITEM hParent,CString strValue)
{
	HTREEITEM hItem=m_ctrlTree.GetChildItem(hParent);
	while(hItem)
	{
		CString strItemText = m_ctrlTree.GetItemText(hItem);
		if(strItemText == strValue)
	return hItem;	
		hItem = m_ctrlTree.GetNextSiblingItem(hItem);
	}

	return m_ctrlTree.InsertItem(strValue,1,2,hParent,TVI_LAST);
}
void CDataDlg::AddListItems()
{
	
	int nCount=0;
	CHeaderCtrl *pHeaderCtrl=m_ctrlList.GetHeaderCtrl();
	if(pHeaderCtrl!=NULL)
		nCount=pHeaderCtrl->GetItemCount();
	for(int i=0;i<nCount;i++)
		m_ctrlList.DeleteColumn(0);
	m_ctrlList.DeleteAllItems();

	for(i=0;i<m_pRS.nFieldCols;i++)
	{
		m_ctrlList.InsertColumn(i,m_pRS.GetFieldName(i),LVCFMT_LEFT,110);
	}

	int nItem=0;
	while(!m_pRS.ADOEOF()){
		m_ctrlList.InsertItem(nItem,m_pRS.GetFieldString(0));
		for(int i=1;i<m_pRS.nFieldCols;i++)
			m_ctrlList.SetItemText(nItem,i,m_pRS.GetFieldString(i));	
		m_pRS.MoveNext();
	}
}


BOOL CDataDlg::OnInitDialog() 
{
	CDialog::OnInitDialog();
	
	// TODO: Add extra initialization here
	CRect rect;
	GetClientRect(rect);
	 // return TRUE unless you set the focus to a control
 	  // EXCEPTION: OCX Property Pages should return FALSE
	m_wndStatusBar.Create(this);
	m_wndStatusBar.SetIndicators(indicators,sizeof(indicators)/sizeof(UINT));
	m_wndStatusBar.MoveWindow(0,rect.bottom-20,rect.right,20);
	m_wndStatusBar.SetPaneStyle(2,SBPS_STRETCH);
	SetTimer(1,1000,NULL);
	
	m_ctrlTree.SetImageList(&m_TreeBootImage,TVSIL_NORMAL);	

	m_ctrlList.ModifyStyle(0,LVS_REPORT|LVS_SHOWSELALWAYS|LVS_SINGLESEL);
	m_ctrlList.SetExtendedStyle(LVS_EX_FULLROWSELECT|LVS_EX_GRIDLINES|LVS_EX_HEADERDRAGDROP);	
	m_pRS.ADOOpen();     
		   
	CDataDlg::OnItemRefresh(); 
	return TRUE; 
}

void CDataDlg::OnItemRefresh() 
{
	// TODO: Add your command handler code here
	m_strSQL="SELECT * FROM ClassmateTab";	
	m_pRS.ADOExcute(m_strSQL);
	AddTreeNodes();  
	AddListItems();	
    
	MyRecordSet rs; 	
	rs.ADOOpen(m_pRS.m_strDBType);                   
	rs.ADOExcute("SELECT DISTINCT StuMajor FROM ClassmateTab");   
	while(!rs.ADOEOF())
	{
		m_ctrlMajor.AddString(rs.GetFieldString(0));
		rs.MoveNext();
	}
	rs.ADOConnectionClose();	
}

void CDataDlg::OnSelchangedTree(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_TREEVIEW* pNMTreeView = (NM_TREEVIEW*)pNMHDR;	
	*pResult = 0;
	HTREEITEM hSelItem=m_ctrlTree.GetSelectedItem();	

	HTREEITEM hParentItem=m_ctrlTree.GetParentItem(hSelItem); 	
	
	CString strSql,str,strCondition;	
	strSql="SELECT * FROM classmateTab";
	if(hParentItem!=NULL)
	{
		HTREEITEM hOldParent=m_ctrlTree.GetParentItem(hParentItem);
		if(hOldParent!=NULL)
		{
			
			strCondition.Format(" Where StuNativePlace='%s' AND StuUniver='%s'",m_ctrlTree.GetItemText(hParentItem),m_ctrlTree.GetItemText(hSelItem));  //此为一行
		}
		else
		{
		
			strCondition.Format(" Where StuNativePlace='%s'",m_ctrlTree.GetItemText(hSelItem));
		}	
	}
	else
	{
		strCondition="";  
	}
	strSql+=strCondition;  
	m_strSQL=strSql;	    
	m_pRS.ADOExcute(strSql);  
	AddListItems();           
}

void CDataDlg::OnClickList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	m_ctrlName.EnableWindow(false);
	int i=m_ctrlList.GetSelectionMark();     

	m_strName=m_ctrlList.GetItemText(i,0);
	m_timeBirthday=m_pRS.StringToTime(m_ctrlList.GetItemText(i,1)); 
	m_strSex=m_ctrlList.GetItemText(i,2);
	m_strUniver=m_ctrlList.GetItemText(i,3);
	m_strNativePlace=m_ctrlList.GetItemText(i,4);
	m_strTel=m_ctrlList.GetItemText(i,5);
	m_strMajor=m_ctrlList.GetItemText(i,6);
	UpdateData(false);	  
	*pResult = 0;
}

void CDataDlg::OnColumnclickList(NMHDR* pNMHDR, LRESULT* pResult) 
{
	NM_LISTVIEW* pNMListView = (NM_LISTVIEW*)pNMHDR;
	// TODO: Add your control notification handler code here
	
	*pResult = 0;
	CPoint   pt;   
	::GetCursorPos(&pt);
	m_ctrlList.ScreenToClient(&pt);   
	LVHITTESTINFO Info;   
	Info.pt = pt;   
	Info.flags = LVHT_ABOVE;
	m_ctrlList.SubItemHitTest(&Info);	

	LVCOLUMN   col;	
	col.mask=LVCF_TEXT;   
	col.cchTextMax =20;
	char   buff[20];   
	col.pszText=buff;
	m_ctrlList.GetColumn(Info.iSubItem,&col);	
	
	CString strSQL=m_strSQL+" ORDER BY "+col.pszText;
	m_pRS.ADOExcute(strSQL);   
	AddListItems();              
}

void CDataDlg::OnItemSearch() 
{
	// TODO: Add your control notification handler code here
	UpdateData(true);	
	if (!m_strName.IsEmpty()) {
	m_pRS.ADOExcute("SELECT * FROM ClassmateTab WHERE StuName='"+m_strName+"'");
	AddListItems();
	UpdateData(false);
	}
	else
		MessageBox("请输入名字","同学查找");
}

void CDataDlg::OnItemClear() 
{
	// TODO: Add your command handler code here
	m_ctrlName.EnableWindow(true);
	m_strName="";
	m_strNativePlace="";
	m_strMajor="";
	m_strUniver="";
	m_strSex="";
	m_strTel="";
	UpdateData(false);
}

void CDataDlg::OnItemDelete() 
{
	// TODO: Add your control notification handler code here
	if(m_ctrlList.GetSelectionMark()<0)
	{
		AfxMessageBox("请先选中要删除的项!");
		return;
	}
	if(AfxMessageBox("您确认要删除吗?",MB_OKCANCEL)==IDOK)
	{
		UpdateData(true);
		CString sql;
		sql.Format("DELETE * FROM ClassmateTab WHERE StuName='%s'",m_strName);
		if(m_pRS.ADOExcuteNoQuery(sql)==1)
		{	
			AfxMessageBox("删除记录成功!");			
			m_ctrlList.DeleteItem(m_ctrlList.GetSelectionMark());
		}
		else
		{
			AfxMessageBox("删除记录失败!");
		}
	}
}

void CDataDlg::OnItemEdit() 
{
	// TODO: Add your control notification handler code here
	UpdateData(true);
	CString sql;
	sql.Format("UPDATE ClassmateTab SET StuSex='%s',StuBirthday='%s',StuNativePlace='%s',StuMajor='%s',StuUniver='%s',StuTel='%s'  WHERE StuName='%s'", 	
		m_strSex,m_pRS.TimeToString(m_timeBirthday),m_strNativePlace,m_strMajor,m_strUniver,m_strTel,m_strName);    //此处为一行
	if (!m_strName.IsEmpty())  {
	if(m_pRS.ADOExcuteNoQuery(sql)==1)
	{	
		AfxMessageBox("编辑记录成功!");
		m_pRS.ADOExcute(m_strSQL);  
		AddListItems();
	}
	}
	else
	{
		AfxMessageBox("编辑记录失败!");
	}
}

void CDataDlg::OnItemHelp() 
{
	// TODO: Add your command handler code here
	MessageBox("HELP");
}

void CDataDlg::OnItemQuit() 
{
	// TODO: Add your command handler code here
	CDialog::OnOK();
}

void CDataDlg::OnTimer(UINT nIDEvent) 
{
	// TODO: Add your message handler code here and/or call default
	CTime time;
	time=CTime::GetCurrentTime();
	CString s=time.Format("%H:%M:%S");
	m_wndStatusBar.SetPaneText(2,s);
	CDialog::OnTimer(nIDEvent);
}


void CDataDlg::OnItemVersion() 
{
	// TODO: Add your command handler code here
	CAboutDialog dlg;
	dlg.DoModal();
}

void CDataDlg::OnItemAdd() 
{
	// TODO: Add your control notification handler code here
	UpdateData(true);
	CString sql;
	sql.Format("INSERT INTO ClassmateTab(StuName,StuBirthday,StuSex,StuUniver,StuNativePlace,StuTel,StuMajor) Values('%s','%s','%s','%s','%s','%s','%s')",
	m_strName,m_pRS.TimeToString(m_timeBirthday),m_strSex,m_strUniver,m_strNativePlace,m_strTel,m_strMajor);               //从sql开始此处为一行
	if (!m_strName.IsEmpty())  {
	if(m_pRS.ADOExcuteNoQuery(sql)==1)
	{	
		AfxMessageBox("增加记录成功!");
	
		int n=m_ctrlList.GetItemCount();
		n=m_ctrlList.InsertItem(n,m_strName);			
		m_ctrlList.SetItemText(n,1,m_pRS.TimeToString(m_timeBirthday));
		m_ctrlList.SetItemText(n,2,m_strSex);
		m_ctrlList.SetItemText(n,3,m_strUniver);
		m_ctrlList.SetItemText(n,4,m_strNativePlace);
		m_ctrlList.SetItemText(n,5,m_strTel);
		m_ctrlList.SetItemText(n,6,m_strMajor);
	}
	else
	{
		AfxMessageBox("增加记录失败,可能是重复的姓名!");
	}}
	else
	{
		AfxMessageBox("增加记录失败,请输入资料!");
	}
}

void CDataDlg::OnContextMenu(CWnd* pWnd, CPoint point) 
{
	// TODO: Add your message handler code here
	CRect rect;
	GetClientRect(&rect);
	CMenu menu;
	menu.LoadMenu(IDR_MENU_DATA);
	CMenu* pConTextMenu=menu.GetSubMenu(0);
	GetCursorPos(&point);
	pConTextMenu->TrackPopupMenu(TPM_LEFTALIGN|TPM_LEFTBUTTON,point.x,point.y,this,&rect);
	menu.DestroyMenu();
}


void CDataDlg::OnButtonBirthday() 
{
	// TODO: Add your control notification handler code here
	CTime time=CTime::GetCurrentTime();
	CString str;
	if (time.GetDay()<10)
		str.Format("%d-0%d",time.GetMonth(),time.GetDay());
	else
		str.Format("%d-%d",time.GetMonth(),time.GetDay());
	m_pRS.ADOExcute("SELECT * FROM ClassmateTab WHERE StuBirthday LIKE'%"+str+"%'");
	AddListItems();
	str="";
	int n=m_ctrlList.GetItemCount();
	for (int i=0;i<n;i++)
		str+=m_ctrlList.GetItemText(i,0)+"  ";
	if (n>0)
		MessageBox("祝  "+str+"生日快乐","生日祝福");
	else
		AfxMessageBox("今天没有同学过生日喔！");
	UpdateData(false);
}